package network2023;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.SocketException;

/**
 * @author 兴趣使然黄小黄
 * @version 1.0
 * @date 2023/4/24 21:12
 * 简单的回显程序服务器
 */
public class UdpEchoServer {

    // 定义一个 socket 对象
    private DatagramSocket socket = null;

    public UdpEchoServer(int port) throws SocketException {
        // 构造 socket 的同时关联/绑定一个端口
        // 不一定绑定成功: 同一主机上一个端口同一时刻只能被一个进程绑定~
        socket = new DatagramSocket(port);
    }

    // 启动服务器
    public void start() throws IOException {
        System.out.println("服务器启动!");
        while (true) {
            // 1. 读取请求并解析
            DatagramPacket requestPacket = new DatagramPacket(new byte[4096], 4096);
            socket.receive(requestPacket); // 如果客户端没有发送数据, receive 就会阻塞
            // 为了方便处理, 将数据报转成 String
            String request = new String(requestPacket.getData(), 0, requestPacket.getLength());
            // 2. 根据请求计算响应(本程序只是回显, 所以不需要额外的计算响应)
            String response = process(request);
            // 3. 将响应结果写回客户端
            DatagramPacket responsePacket =
                    new DatagramPacket(response.getBytes(), response.getBytes().length, requestPacket.getSocketAddress());
            socket.send(responsePacket);
            System.out.printf("[%s:%d] req: %s, resp: %s\n",
                    requestPacket.getAddress().toString(), requestPacket.getPort(), request, response);
        }
    }

    // 根据请求计算响应, 此处只是回显
    public String process(String request) {
        return request;
    }

    public static void main(String[] args) throws IOException {
        UdpEchoServer udpEchoServer = new UdpEchoServer(9090);
        udpEchoServer.start();
    }
}
